package com.eduction.common;


import org.apache.commons.io.FileUtils;

import org.springframework.util.ResourceUtils;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;



public class ZipDownloadUtil {



    protected static String getBaseUrl() {
        String baseUrl = getRequest().getScheme() + "://" + getRequest().getServerName()
                + ":" + getRequest().getServerPort()
                + getRequest().getServletPath();
        return baseUrl;
    }

    protected static HttpServletRequest getRequest() {
        ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        return servletRequestAttributes.getRequest();
    }
    public static void cleanPath(HttpServletRequest request)throws IOException{
        String tmpPath = request.getSession().getServletContext().getRealPath("/")+ TimeUtil.getCurrentTime(TimeUtil.FORMAT_DATE)+File.separator;
        FileUtils.cleanDirectory(new File(tmpPath));
    }


    /**
     *
     * @param filePathList 文件地址集合
     * @param zipFileName zip包名称
     * @param request
     * @param response
     * @throws IOException
     */
    public static void downloadZip(List<String> filePathList, String zipFileName, HttpServletRequest request, HttpServletResponse response)throws IOException{
        if(filePathList!=null&&!filePathList.isEmpty()){
            String tmpPath = request.getSession().getServletContext().getRealPath("/") + TimeUtil.getCurrentTime(TimeUtil.FORMAT_DATE) + File.separator;

            FileUtils.forceMkdir(new File(tmpPath));
            List<File> files = new ArrayList<File>();
            for(String filePath:filePathList){
                File file=new File(filePath);
                files.add(file);
            }
            zipFileName = zipFileName + ".zip";
            // 在服务器端创建打包下载的临时文件
            String outFilePath = tmpPath;

            File fileZip = new File(outFilePath + zipFileName);
            // 文件输出流
            FileOutputStream outStream = new FileOutputStream(fileZip);
            // 压缩流
            ZipOutputStream toClient = new ZipOutputStream(outStream);
            //  toClient.setEncoding("gbk");
            zipFile(files, toClient);
            toClient.close();
            outStream.close();
            downloadFile(fileZip, request,response, true);

        }


    }
    public static void zipFile(List<File> files, ZipOutputStream outputStream) throws IOException{
        try {
            int size = files.size();
            // 压缩列表中的文件
            for (int i = 0; i < size; i++) {
                File file = (File) files.get(i);
                zipFile(file, outputStream);
            }
        } catch (IOException e) {
            throw e;
        }
    }
    public static void zipFile(File inputFile, ZipOutputStream outputstream) throws IOException {
        try {
            if (inputFile.exists()) {
                if (inputFile.isFile()) {
                    FileInputStream inStream = new FileInputStream(inputFile);
                    BufferedInputStream bInStream = new BufferedInputStream(inStream);
                    ZipEntry entry = new ZipEntry(inputFile.getName());
                    outputstream.putNextEntry(entry);

                    final int MAX_BYTE = 10 * 1024 * 1024; // 最大的流为10M
                    long streamTotal = 0; // 接受流的容量
                    int streamNum = 0; // 流需要分开的数量
                    int leaveByte = 0; // 文件剩下的字符数
                    byte[] inOutbyte; // byte数组接受文件的数据

                    streamTotal = bInStream.available(); // 通过available方法取得流的最大字符数
                    streamNum = (int) Math.floor(streamTotal / MAX_BYTE); // 取得流文件需要分开的数量
                    leaveByte = (int) streamTotal % MAX_BYTE; // 分开文件之后,剩余的数量

                    if (streamNum > 0) {
                        for (int j = 0; j < streamNum; ++j) {
                            inOutbyte = new byte[MAX_BYTE];
                            // 读入流,保存在byte数组
                            bInStream.read(inOutbyte, 0, MAX_BYTE);
                            outputstream.write(inOutbyte, 0, MAX_BYTE); // 写出流
                        }
                    }
                    // 写出剩下的流数据
                    inOutbyte = new byte[leaveByte];
                    bInStream.read(inOutbyte, 0, leaveByte);
                    outputstream.write(inOutbyte);
                    outputstream.closeEntry(); // Closes the current ZIP entry
                    // and positions the stream for
                    // writing the next entry
                    bInStream.close(); // 关闭
                    inStream.close();
                }
            } else {
                throw new IOException("文件不存在！");
            }
        } catch (IOException e) {
            throw e;
        }
    }


    public static void downloadFile(File file, HttpServletRequest request, HttpServletResponse response, boolean isDelete) {
        try {
            BufferedInputStream fis = new BufferedInputStream(new FileInputStream(file.getPath()));
            byte[] buffer = new byte[fis.available()];
            fis.read(buffer);
            fis.close();
            response.reset();
            OutputStream toClient = new BufferedOutputStream(response.getOutputStream());
            response.setContentType("application/octet-stream");
            response.setHeader("Content-Disposition", "attachment;filename=" + new String(file.getName().getBytes("UTF-8"),"ISO-8859-1"));
            toClient.write(buffer);
            toClient.flush();
            toClient.close();
            if(isDelete)
            {
                file.delete();        //是否将生成的服务器端文件删除
            }
            cleanPath(request);
        }catch (IOException ex){
            ex.printStackTrace();
        }
    }

}  